package com.dexter.year2023.charpter8_tree_hot_problems.level1.topic1_2路径;

import com.dexter.year2023.charpter6_tree_level_travel.level1.TreeNode;

import java.util.ArrayList;
import java.util.List;

/**
 * 257. 二叉树的所有路径
 * https://leetcode.cn/problems/binary-tree-paths/
 * <p>
 * 给你一个二叉树的根节点 root ，按 任意顺序 ，返回所有从根节点到叶子节点的路径。
 * 叶子节点 是指没有子节点的节点。
 * <p>
 * 示例 1：
 * 输入：root = [1,2,3,null,5]
 * 输出：["1->2->5","1->3"]
 * 示例 2：
 * 输入：root = [1]
 * 输出：["1"]
 * <p>
 * 提示：
 * 树中节点的数目在范围 [1, 100] 内
 * -100 <= Node.val <= 100
 *
 * @author Dexter
 */
public class LC257BinaryTreePaths {
    public static void main(String[] args) {
        TreeNode node = new TreeNode(3);
        node.left = new TreeNode(9);
        node.right = new TreeNode(20);
        node.right.left = new TreeNode(15);
        node.right.right = new TreeNode(7);
        // 提交结果：已通过 - 时间击败14.11% 的Java用户，内存击败29.51% 的Java用户
        System.out.println(binaryTreePaths(node));
    }

    public static List<String> binaryTreePaths(TreeNode root) {
        List<String> res = new ArrayList<>();
        dfs(root, "", res);
        return res;
    }

    static void dfs(TreeNode root, String path, List<String> res) {
        // 终止条件
        if (root == null) {
            return;
        }
        // 找到叶子节点
        if (root.left == null && root.right == null) {
            res.add(path + root.val);
            return;
        }
        // 递归，左、右节点
        dfs(root.left, path + root.val + "->", res);
        dfs(root.right, path + root.val + "->", res);
    }

    /**
     * StringBuilder优化写法 - 时间击败99.99% 的Java用户，内存击败72.54% 的Java用户
     *
     * @param root 每次判断的节点
     * @param path 路径字符串
     * @param res  字符串集合
     */
    static void dfs2(TreeNode root, String path, List<String> res) {
        if (root != null) {
            StringBuilder sb = new StringBuilder(path);
            sb.append(root.val);
            // 终止条件
            if (root.left == null && root.right == null) {
                // 该路径走到叶子节点了
                res.add(sb.toString());
                return;
            }
            // 递归，左、右节点
            sb.append("->");
            dfs2(root.left, sb.toString(), res);
            dfs2(root.right, sb.toString(), res);
        }
    }
}
